pytorch custom dataset padding